from .temperature_scaling import ModelWithTemperature